from . import DEFAULT_MODEL_NAME, Prompt, client

USER_PROMPT = """
以下是我的具体问题：
{question}

以下是有关这个问题的几个标准答案：
{context}

请将上述内容聚合，回答我的问题。注意不要遗漏信息，要包含所有的关键信息，特别注意如果有注意事项及温馨提示等内容，要特别强调，切勿遗漏。
"""


def build_prompt(question: str, context: str) -> Prompt:
    user_prompt = USER_PROMPT.format(question=question, context=context)
    return Prompt(system_prompt="", user_prompt=user_prompt)


async def run_agent(question: str, context: str) -> str:
    prompt: Prompt = build_prompt(question, context)
    chat_completion = await client.chat.completions.create(
        messages=[
            {"role": "system", "content": ""},
            {"role": "user", "content": prompt.user_prompt},
        ],
        model=DEFAULT_MODEL_NAME,
    )
    # get response str
    response = chat_completion.choices[0].message.content
    return response
